#验证码工具类
#导入随机库

import random
#导入redis库
import redis
from db import Db
import datetime
import time

#定义验证码类
class Code:
    def __init__(self):
        #声明resdis连接
        self.db = Db()
        self.r = redis.Redis()
        code = ''
        #定义随机字符
        source = '0123456789'
        for i in range(4):
            #获取随机码
            tmp_num = random.randrange(len(source))
            code += source[tmp_num]
        self.code= code

    #发送验证码
    def send_code(self,phone):
        #判断两种情况
        code = self.r.get(phone)
        if code:
            return '你30秒内只能发送一条验证码'
        else:
            #存储验证码
            self.r.set(phone,self.code)
            #生命周期
            self.r.expire(phone,30)
            return self.code


